"""
@Time: 2020/12/17 上午 9:51
@Author: jinzhuan
@File: ace2005.py
@Desc: 
"""
import torch
from cognlp.io.loader.ee.ace2005 import ACE2005Loader
from cognlp.io.processor.ee.ace2005 import ACE2005Processor
from cognlp.core.datableset import DataTableSet

torch.cuda.set_device(1)
loader = ACE2005Loader()
train_data, dev_data, test_data = loader.load_all('../../../../cognlp/data/ee/ace2005/data')
processor = ACE2005Processor(trigger_path='../../../../cognlp/data/ee/ace2005/vocabulary.txt',
                             argument_path='../../../../cognlp/data/ee/ace2005/argument_vocabulary.txt')
datable = processor.process(test_data)
dataset = DataTableSet(datable)
dataset.__getitem__(6)
datable.split(1, 1, 1)
dataset.__getitem__(0)
